package com.z.redis.command.impl.zset;

import com.z.redis.RedisCore;
import com.z.redis.command.Command;
import com.z.redis.command.CommandType;
import com.z.redis.datatype.BytesWrapper;
import com.z.redis.datatype.RedisData;
import com.z.redis.datatype.RedisZset;
import com.z.redis.resp.BulkString;
import com.z.redis.resp.Resp;
import com.z.redis.resp.RespInt;
import io.netty.channel.ChannelHandlerContext;

/**
 * zcard key
 * 获取有序集合的成员数
 */
public class Zcard implements Command {
    private BytesWrapper key;

    @Override
    public CommandType type(){
        return CommandType.zcard;
    }

    @Override
    public void setContent(Resp[] array){
        key = ((BulkString) array[1]).getContent();
    }

    @Override
    public void handle(ChannelHandlerContext ctx, RedisCore redisCore){
        RedisData redisData = redisCore.get(key);
        if(redisData == null){
            ctx.writeAndFlush(new RespInt(0));
        }else if(redisData instanceof RedisZset){
            int zcard = ((RedisZset) redisData).zcard();
            ctx.writeAndFlush(new RespInt(zcard));
        }else{
            throw new UnsupportedOperationException("type not Zset");
        }
    }
}
